package csu.web.jpetstore.service;

import java.awt.*;
import java.awt.image.BufferedImage;
import java.util.Random;

public class VerificationService {
    private int width = 80; // 验证码图片的长和宽
    private int height = 30;
    private String text; // 用来保存验证码的文本内容
    private Random random = new Random(); // 获取随机数对象
    private String[] fontNames = { "宋体", "华文楷体", "黑体", "微软雅黑", "楷体_GB2312" }; // 字体数组
    private String codes = "23456789abcdefghjkmnopqrstuvwxyzABCDEFGHJKMNPQRSTUVWXYZ"; // 验证码数组

    /**
     * 获取随机的颜色
     * @return Color(r,g,b)
     */
    private Color randomColor() {
        int r = this.random.nextInt(200); // 这里为什么是150，因为当r，g，b都为255时，即为白色，为了好辨认，需要颜色深一点。
        int g = this.random.nextInt(200);
        int b = this.random.nextInt(200);
        return new Color(r, g, b); // 返回一个随机颜色
    }

    private Color getColor(int start,int end){

        int r=start+this.random.nextInt(end-start);
        int g=start+this.random.nextInt(end-start);
        int b=start+this.random.nextInt(end-start);

        return new Color(r,g,b);
    }
    private Font randomFont() // 获取随机字体
    {
        int index = random.nextInt(fontNames.length); // 获取随机的字体
        String fontName = fontNames[index];
        int style = random.nextInt(4); // 随机获取字体的样式，0是无样式，1是加粗，2是斜体，3是加粗加斜体
        int size = random.nextInt(5) + 24; // 随机获取字体的大小
        return new Font(fontName, style, size); // 返回一个随机的字体
    }

    private char randomChar() // 获取随机字符
    {
        int index = random.nextInt(codes.length());
        return codes.charAt(index);
    }

    private void drawLine(BufferedImage image) // 画干扰线，验证码干扰线用来防止计算机解析图片
    {
        int num = 5; // 定义干扰线的数量
        //获取指向该图像上的Graphic画笔
        Graphics2D g = (Graphics2D) image.getGraphics();
        for (int i = 0; i < num; i++) {
            int x1 = random.nextInt(width);
            int y1 = random.nextInt(height);
            int x2 = random.nextInt(width);
            int y2 = random.nextInt(height);
            g.setColor(getColor(0,180));
            g.drawLine(x1, y1, x2, y2);
        }
    }

    private BufferedImage createImage() // 创建图片
    {
        BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB); // 创建图片缓冲区
        Graphics2D g = (Graphics2D) image.getGraphics(); // 获取画笔
        g.setColor(getColor(150, 255)); // 设置背景色
        g.fillRect(0, 0, width, height);
        return image; // 返回一个图片
    }

    public BufferedImage getImage() // 获取验证码图片的方法
    {
        BufferedImage image = createImage();
        Graphics2D g = (Graphics2D) image.getGraphics(); // 获取画笔
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < 4; i++) // 画四个字符即可
        {
            String s = randomChar() + ""; // 随机生成字符，因为只有画字符串的方法，没有画字符的方法，所以需要将字符变成字符串再画
            sb.append(s); // 添加到StringBuilder里面
            float x = i * 1.0F * width / 4; // 定义字符的x坐标
            g.setFont(randomFont()); // 设置字体，随机
            g.setColor(getColor(0,180)); // 设置颜色，随机
            g.drawString(s, x, height - 5);
        }
        this.text = sb.toString();
        drawLine(image);
        return image;
    }

    public String getText() // 获取验证码文本的方法
    {
        return text;
    }
}
